package tools;

import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;

import javax.servlet.http.HttpServletRequest;

public class FileTool {

	public static String downLoadFromUrl(String urlStr, String fileName,
			String savePath) {
		try {
			URL url = new URL(urlStr);
			HttpURLConnection conn = (HttpURLConnection) url.openConnection();
			
			// 设置超时间为3秒
			conn.setConnectTimeout(3 * 1000);

			// 得到输入流
			InputStream inputStream = conn.getInputStream();
			// 获取自己数组
			byte[] getData = readInputStream(inputStream);

			// 文件保存位置
			File saveDir = new File(savePath);
			if (!saveDir.exists()) {
				saveDir.mkdir();
			}

			// fileName=new String(fileName.getBytes("iso8859-1"),"UTF-8");
			File file = new File(saveDir + File.separator + fileName);
			FileOutputStream fos = new FileOutputStream(file);
			fos.write(getData);
			if (fos != null) {
				fos.close();
			}
			if (inputStream != null) {
				inputStream.close();
			}
			// System.out.println("info:"+url+" download success");
			return saveDir + File.separator + fileName;
		} catch (Exception e) {
			e.printStackTrace();
		}
		return "";

	}

	/**
	 * 从输入流中获取字节数组
	 * 
	 * @param inputStream
	 * @return
	 * @throws IOException
	 */
	public static byte[] readInputStream(InputStream inputStream)
			throws IOException {
		byte[] buffer = new byte[1024];
		int len = 0;
		ByteArrayOutputStream bos = new ByteArrayOutputStream();
		while ((len = inputStream.read(buffer)) != -1) {
			bos.write(buffer, 0, len);
		}
		bos.close();
		return bos.toByteArray();
	}

	/**
	 * @Title: getBrowser
	 * @Description: 判断客户端浏览器
	 * @return String
	 * @author
	 * @date
	 */
	public static String getBrowser(HttpServletRequest request) {
		String UserAgent = request.getHeader("USER-AGENT").toLowerCase();
		if (UserAgent != null) {
			if (UserAgent.indexOf("msie") != -1)
				return "IE";
			if (UserAgent.indexOf("firefox") != -1)
				return "FF";
			if (UserAgent.indexOf("safari") != -1)
				return "SF";
			if (request.getHeader("User-Agent").toUpperCase().indexOf("CHROME") > 0)// 谷歌
				return "CH";
		}
		return null;
	}

	public static void main(String[] args) {
		String url = "http://localhost:8080/upload/tutou.jpg";
		// String url = "http://www.diaiy.com/frontend/web/app/lushi/fl.png";
		String[] items = url.split("\\.");
		String fileType = items[items.length - 1];
		String path = FileTool.downLoadFromUrl(url, "temp." + fileType,
				"D:\\gameImage\\");//
		System.out.println(path);
	}

}